package com.qd.system.service.article;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.cdqidi.constant.ConstantDto;
import com.cdqidi.db.BaseService;
import com.cdqidi.dto.CacheKey;
import com.cdqidi.exception.ApiException;
import com.cdqidi.util.FileUtil;
import com.cdqidi.util.RedisTemplateUtil;
import com.qd.common.sys.domain.entity.article.ArticleFileDTO;
import com.qd.system.mapper.ArticleFileMapper;
import com.qd.system.model.ArticleFile;
import com.qd.system.service.dict.MbUtil;
import com.qd.system.util.FileConfirmUtil;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.StringUtils;

import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.Optional;

import static java.util.stream.Collectors.toList;

/**
 * 附件
 *
 * @author sjk
 */
@RequiredArgsConstructor
@Service
@Slf4j
public class ArticleFileService extends BaseService<ArticleFileMapper, ArticleFile> {

    private static final String PREFIX_ID = "sys_article_file_docId_{0}";
    private static final String PREFIX_LIST = "sys_article_file_articleId_{0}";

    private final RedisTemplateUtil<ArticleFile> redisTemplateUtil;

    /**
     * 保存
     *
     * @param dto 页面对象
     * @return 页面对象
     */
    @Transactional(rollbackFor = Exception.class)
    public ArticleFileDTO save(ArticleFileDTO dto) {
        final ArticleFile model = dtoToModel(dto);
        try {
            if (super.save(model)) {
                return modelToDto(model);
            }
            throw new ApiException("添加失败");
        } finally {
            clearCache(model);
            model.freeData();
        }
    }

    /**
     * 修改
     *
     * @param dto 页面对象
     * @return 页面对象
     */
    @Transactional(rollbackFor = Exception.class)
    public ArticleFileDTO update(ArticleFileDTO dto) {
        final ArticleFile historyModel = Optional.ofNullable(getById(dto.getDocId())).orElseThrow(() -> new ApiException("ID不存在,不能更新"));
        final ArticleFile model = dtoToModel(dto);
        try {
            if (super.updateById(model)) {
                return modelToDto(model);
            }
            throw new ApiException("更新失败");
        } finally {
            clearCache(historyModel);
            historyModel.freeData();
            model.freeData();
        }
    }

    /**
     * 批量添加
     *
     * @param mList 数据库对象
     */
    @Transactional(rollbackFor = Exception.class)
    public void saveBatch(List<ArticleFile> mList) throws ApiException {
        try {
            if (!super.saveBatch(mList)) {
                throw new ApiException("批量添加失败");
            }
        } finally {
            clearCaches(mList);
        }
    }

    /**
     * 删除，单个
     *
     * @param docId 主键
     */
    @Transactional(rollbackFor = Exception.class)
    public void removeById(String docId) {
        final ArticleFile model = Optional.ofNullable(getById(docId)).orElseThrow(() -> new ApiException("数据不存在,不能删除"));
        try {
            if (!super.removeById(docId)) {
                throw new ApiException("删除失败");
            }
        } finally {
            clearCache(model);
            model.freeData();
        }
    }

    /**
     * 批量删除
     *
     * @param ids 主键集合
     */
    @Transactional(rollbackFor = Exception.class)
    public void removeBatchIds(List<String> ids) {
        List<ArticleFile> historyModelList = checkIds(ids);
        try {
            if (!super.removeByIds(ids)) {
                throw new ApiException("批量删除失败");
            }
        } finally {
            clearCaches(historyModelList);
            historyModelList.clear();
        }
    }

    /**
     * 根据传入对象构建查询参数，判断是否存在
     *
     * @param dto 参数
     * @return 存在为true
     */
    @Transactional(readOnly = true)
    public Boolean isExists(ArticleFileDTO dto) {
        final QueryWrapper<ArticleFile> queryWrapper = createQueryWrapper(dto);
        final boolean exists = super.count(queryWrapper) > 0;
        queryWrapper.clear();
        return exists;
    }

    /**
     * 根据主键判断数据是否存在
     *
     * @param docId 主键
     * @return 存在为true
     */
    @Transactional(readOnly = true)
    public Boolean isExistsById(String docId) {
        boolean exists = false;
        if (StringUtils.hasLength(docId)) {
            final ArticleFile model = getById(docId);
            exists = (null != model);
            if (exists) {
                model.freeData();
            }
        }
        return exists;
    }

    /**
     * 根据主键获取数据
     *
     * @param docId 主键
     * @return 单条记录
     */
    @Transactional(readOnly = true)
    public ArticleFile getById(String docId) {
        final String key = RedisTemplateUtil.getRedisCacheKey(PREFIX_ID, docId);
        return redisTemplateUtil.get(new CacheKey().setKey(key).setTimeout(ConstantDto.REDIS_TIME_OUT_DAY), () -> super.getById(docId), ArticleFile.class);
    }

    /**
     * 根据主键获取数据
     *
     * @param docId 主键
     * @return 单条记录，转换成页面对象
     */
    @Transactional(readOnly = true)
    public ArticleFileDTO getByIdDto(String docId) {
        final ArticleFile model = getById(docId);
        if (null != model) {
            return modelToDto(model);
        }
        return null;
    }

    /**
     * 根据条件查询单条数据
     *
     * @param dto 查询参数
     * @return 单条记录
     */
    @Transactional(readOnly = true)
    public ArticleFile getOne(ArticleFileDTO dto) {
        final QueryWrapper<ArticleFile> queryWrapper = createQueryWrapper(dto);
        ArticleFile model = super.getOne(queryWrapper);
        queryWrapper.clear();
        return model;
    }

    /**
     * 根据条件查询单条数据
     *
     * @param dto 查询参数
     * @return 单条记录，转换成页面对象
     */
    @Transactional(readOnly = true)
    public ArticleFileDTO getOneDto(ArticleFileDTO dto) {
        final ArticleFile model = getOne(dto);
        if (null == model) {
            return null;
        }
        final ArticleFileDTO newDto = modelToDto(model);
        model.freeData();
        return newDto;
    }

    /**
     * 根据条件查询列表数据
     *
     * @return 列表数据
     */
    @Transactional(readOnly = true)
    public List<ArticleFile> getList(String articleId) {
        final ArticleFileDTO query = new ArticleFileDTO();
        query.setArticleId(articleId);
        final QueryWrapper<ArticleFile> queryWrapper = createQueryWrapper(query);
        queryWrapper.orderByAsc("idx");
        final String key = RedisTemplateUtil.getRedisCacheKey(PREFIX_LIST, articleId);
        final List<ArticleFile> list = redisTemplateUtil.getList(new CacheKey().setKey(key), () -> super.list(queryWrapper), ArticleFile.class);
        queryWrapper.clear();
        return list;
    }

    @Transactional(readOnly = true)
    public List<ArticleFileDTO> getListDto(String articleId) {
        List<ArticleFile> list = getList(articleId);
        if (null == list || list.isEmpty()) {
            return Collections.emptyList();
        }
        return list.stream().map(this::modelToDto).collect(toList());
    }

    /**
     * 清空缓存
     *
     * @param model 数据库对象
     */
    public void clearCache(ArticleFile model) {
        if (null == model) {
            return;
        }
        redisTemplateUtil.deleteByKey(RedisTemplateUtil.getRedisCacheKey(PREFIX_ID, model.getDocId()));
        if (StringUtils.hasLength(model.getArticleId())) {
            redisTemplateUtil.deleteByKey(RedisTemplateUtil.getRedisCacheKey(PREFIX_LIST, model.getArticleId()));
        }
    }

    /**
     * 批量清空缓存
     *
     * @param list 数据库对象
     */
    public void clearCaches(List<ArticleFile> list) {
        if (null == list || list.isEmpty()) {
            return;
        }
        List<String> ids = list.stream().map(model -> RedisTemplateUtil.getRedisCacheKey(PREFIX_ID, model.getDocId())).collect(toList());
        if (!ids.isEmpty()) {
            redisTemplateUtil.deleteByKey(ids);
            ids.clear();
        }
        List<String> uniqueIds = list.stream().filter(model -> StringUtils.hasLength(model.getArticleId())).map(model -> RedisTemplateUtil.getRedisCacheKey(PREFIX_LIST, model.getArticleId())).collect(toList());
        if (!uniqueIds.isEmpty()) {
            redisTemplateUtil.deleteByKey(uniqueIds);
            uniqueIds.clear();
        }
    }

    /**
     * 数据库对象转页面对象
     *
     * @param model 数据库对象
     * @return 页面对象
     */
    public ArticleFileDTO modelToDto(ArticleFile model) {
        if (null == model) {
            return null;
        }
        final ArticleFileDTO dto = new ArticleFileDTO();
        dto.setDocId(model.getDocId());
        dto.setDocName(model.getDocName());
        dto.setDocPath(model.getDocPath());
        dto.setDocSize(model.getDocSize());
        dto.setIdx(model.getIdx());
        dto.setArticleId(model.getArticleId());
        dto.setAttachmentType(model.getAttachmentType());
        buildDisplay(dto);
        return dto;
    }

    /**
     * 页面对象转数据库对象
     *
     * @param dto 页面对象
     * @return 数据库对象
     */
    public ArticleFile dtoToModel(ArticleFileDTO dto) {
        if (null == dto) {
            return null;
        }
        final ArticleFile model = new ArticleFile();
        model.setDocId(dto.getDocId());
        model.setDocName(dto.getDocName());
        model.setDocPath(dto.getDocPath());
        model.setDocSize(dto.getDocSize());
        model.setIdx(dto.getIdx());
        model.setArticleId(dto.getArticleId());
        model.setAttachmentType(dto.getAttachmentType());
        return model;
    }

    /**
     * 码表转换
     *
     * @param dto 页面对象
     */
    public void buildDisplay(ArticleFileDTO dto) {
        dto.setDocSizeDisplay(FileUtil.getSizeDisplay(dto.getDocSize()));
        dto.setDocPathDisplay(FileConfirmUtil.getFilePath(dto.getDocPath(), dto.getDocName()));
        dto.setAttachmentTypeDisplay(MbUtil.idToDisplay(dto.getAttachmentType()));
    }

    /**
     * 数据库对象转页面对象
     *
     * @param list 数据库对象
     * @return 页面对象
     */
    public List<ArticleFileDTO> modelToDtoList(List<ArticleFile> list) {
        if (null == list) {
            return Collections.emptyList();
        }
        return list.stream().map(this::modelToDto).collect(toList());
    }

    /**
     * 页面对象转数据库对象
     *
     * @param list 页面对象
     * @return 数据库对象
     */
    public List<ArticleFile> dtoToModelList(List<ArticleFileDTO> list) {
        if (null == list) {
            return Collections.emptyList();
        }
        return list.stream().map(this::dtoToModel).collect(toList());
    }

    /**
     * 拼装查询参数，除了主键的其他字段会自动加入进来
     *
     * @param dto 查询参数，不为空的属性会被自动加入
     */
    public QueryWrapper<ArticleFile> createQueryWrapper(ArticleFileDTO dto) {
        return createQueryWrapper(dto, "");
    }

    /**
     * 拼装查询参数，除了主键的其他字段会自动加入进来
     *
     * @param dto        查询参数，不为空的属性会被自动加入
     * @param tableAlias 表别名
     */
    public QueryWrapper<ArticleFile> createQueryWrapper(ArticleFileDTO dto, String tableAlias) {
        final QueryWrapper<ArticleFile> queryWrapper = new QueryWrapper<>();
        if (null != dto) {
            if (StringUtils.hasLength(tableAlias) && tableAlias.charAt(tableAlias.length() - 1) != '.') {
                tableAlias += ".";
            }
            if (StringUtils.hasLength(dto.getDocName())) {
                queryWrapper.eq(tableAlias + "doc_name", dto.getDocName());
            }
            if (StringUtils.hasLength(dto.getDocPath())) {
                queryWrapper.eq(tableAlias + "doc_path", dto.getDocPath());
            }
            if (StringUtils.hasLength(dto.getArticleId())) {
                queryWrapper.eq(tableAlias + "article_id", dto.getArticleId());
            }
        }
        return queryWrapper;
    }

    /**
     * 批量检查主键在数据库中是否存在
     *
     * @param ids 主键集合
     * @return 主键查询出来的集合
     */
    public List<ArticleFile> checkIds(List<String> ids) {
        if (null == ids || ids.isEmpty()) {
            throw new ApiException("ids不能为空");
        }
        final List<ArticleFile> list = super.listByIds(ids);
        if (list.isEmpty()) {
            throw new ApiException("ids不存在" + Arrays.toString(ids.toArray()));
        }
        if (ids.size() == list.size()) {
            return list;
        }
        List<String> dbIds = list.stream().map(ArticleFile::getDocId).collect(toList());
        final List<String> notExistsList = ids.stream().filter(id -> !dbIds.contains(id)).collect(toList());
        if (!notExistsList.isEmpty()) {
            throw new ApiException("ids不存在" + Arrays.toString(notExistsList.toArray()));
        }
        return list;
    }

    @Transactional(rollbackFor = Exception.class)
    public void removeByIds(List<String> ids) {
        List<ArticleFile> articleFiles = super.getBaseMapper().selectBatchIds(ids);
        ids.forEach(this::removeById);
        articleFiles.forEach(articleFile -> FileConfirmUtil.deleteFileConfirm(articleFile.getDocPath()));
    }
}